import pandas as pd


def read_data(file_name):
    """
    读取文件
    :param file_name: 读取对象
    :return: parameter_list: 为pytest参数化提供数据对象
    注：必须将读取对象放到data文件夹下
    """
    parameter_list = []
    pd.set_option('display.max_columns', None)  # 显示所有列
    pd.set_option('display.max_rows', None)  # 显示所有行
    pd.set_option('max_colwidth', 100)  # 设置value的显示长度为100，默认为50
    df = pd.read_excel(r'../data/{}'.format(file_name))  # 这个会直接默认读取到这个Excel的第一个表单
    for i in range(len(df)):
        data = df.loc[i]  # 读取第一行的数据
        data_to_list = [j for j in data]  # 将data进行list转换方便操作
        tuples = (data_to_list[1], data_to_list[2], data_to_list[3], data_to_list[4])  # 为pytest参数化构造数据元组
        parameter_list.append(tuples)
    return parameter_list


if __name__ == '__main__':
    datas = read_data("demo.xlsx")
    print(datas)
